Optimizing What Policies Learn From: Recoverability-aware Rollout Intervention Learning
可恢复性感知的 Rollout 干预学习
🧭 LLM Agent 后训练 · 可恢复性感知的 rollout 干预 ★ 深度解读
University of Notre Dame · Amazon
将 rollout 干预建模为在线 bandit,从 reward contrast 增益学习在何处、如何分支,提供更强的训练信号并节省算力
📌 工作参考价值(4 点)
- 用 reward contrast 增益代替固定启发式作为干预信号:在对话推荐或商品搜索 Agent 训练中,不用固定的熵或成功率来分配分支预算,而是预测一次额外 rollout 能带来的 reward 方差增益(recoverability),并将其作为在线 bandit 的反馈,学习动态分支策略。
- Shadow-to-Live 双阶段部署:先用高熵 anchor 检测 + 迭代分支收集 (state, action, gain) 轨迹,训练一个轻量控制器;在线训练时用 utility gate 过滤低增益分支,并用 recency‑weighted 滑动窗口持续更新控制器。这种「先暖场再上场」的范式可直接迁移到线上 Agent 训练系统。
- 结构化干预空间:不要只调 rollout 数量,同时联合选择 branch budget 和 decoding 温度(exploit / explore),让控制器学到「增加采样数」和「增加采样多样性」的组合策略,在稀疏奖励的交互推荐场景中尤其有效。
- 在训练流程中实现算力与效果的 Pareto 改善:实验表明 RAIL 用不到半数的 rollout 即可超越 GRPO‑32,在成本敏感的线上推荐 Agent 训练中,可以用更少的交互获得更好的策略,显著降低环境交互开销。